# 导入第三方库
import os
import sys

# 导入内置库
from main_modules.functions import MyJson
from main_modules.functions import Shell

# 检测是否设置相关信息

# 获取当前工作目录的绝对路径
# root_path = os.getcwd()
root_path = "F:\Mys"

# 获取当前目录的文件
root_things = os.listdir(root_path)

# 获取My.json的绝对路径
myJson_path = os.path.join(root_path, "My.json")

# 函数部分
def yeMyJson(myJson_path):
    my_json = MyJson.readMyJson(myJson_path)
    print("%s欢迎回家!!!" % my_json["username"])
    exit = 3
    while True:
        if (input("请输入密码:") == my_json["password"]):
            print("密码正确, 正在进入系统...")
            break
        elif exit == 1:
            print("密码错误三次，退出系统")
            sys.exit()
        else:
            exit -= 1
            print('密码错误')
    return [my_json["username"], my_json["password"]]


def noMyJson(myJson_path):
    print("宁是第一次使用，请设置相关信息")
    while True:
        username = input("请输入用户名: ")
        if username.strip() == "":
            print("用户名不能为空!!!")
            continue
        break
    while True:
        password = input("请输入密码: ")
        if password.strip() == "":
            print("密码不能为空!!!")
            continue
        break

    if (MyJson.NewMyJson(username, password, myJson_path)):
        print("初始化成功")
        return True
    else:
        print("初始化失败，请检查程序是否有误")
        return False

# 如果有My.json
if "My.json" in root_things:
    info = yeMyJson(myJson_path)
    Shell.shell(info[0], info[1], root_path)
# 如果无My.json, 第一次使用，进行配置
else:
    noMyJson(myJson_path)
    info = yeMyJson(myJson_path)
    Shell.shell(info[0], info[1], root_path)


    
